package sword_offer;

import common.TreeNode;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 * @ClassName _68_1_lowestCommonAncestor
 * @Description: 给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
 *
 * 百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”
 *
 * 例如，给定如下二叉树:  root = [3,5,1,6,2,0,8,null,null,7,4]
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/er-cha-shu-de-zui-jin-gong-gong-zu-xian-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * @Author: yongliang.ma
 * @Create_time:2022/10/11 12:38
 */
public class _68_2_lowestCommonAncestor {

    private Map<Integer, TreeNode> parentMap = new HashMap<>();

    private Set<TreeNode> visited = new HashSet<>();

    // 存储父节点值
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        dfs(root, null);

        // 遍历p，存储其历史记录
        TreeNode pCursor = p;
        while (pCursor != null){
            visited.add(pCursor);
            pCursor = parentMap.getOrDefault(pCursor.val, null);
        }

        // 遍历节点q
        while (q != null){
            if (visited.contains(q))
                return q;

            q = parentMap.getOrDefault(q.val, null);
        }
        return null;
    }

    private void dfs(TreeNode node, TreeNode parentNode){
        if (node == null)
            return;
        parentMap.put(node.val, parentNode);
        if (node.left != null)
            dfs(node.left, node);
        if (node.right != null)
            dfs(node.right, node);
    }
}
